python color labelling